from django.db import models

# Create your models here.
from django.db.models.query import QuerySet
from django.utils import timezone 
from django.contrib.auth.models import User 
from django.urls import reverse
from taggit.managers import TaggableManager
# 类名PublishedManager为⾃拟名称，该类继承⾃models.Manager
class PublishedManager(models.Manager):
 # 重写Manager⽗类的get_queryset⽅法，'-> QuerySet'是新语法表示⽅法的返回值是QuerySet类型
 def get_queryset(self) -> QuerySet:
 # 通过super()调⽤⽗类get_queryset()⽅法获取所有⽂章再调⽤过滤⽅法filter，获得已经发布的⽂章
   return super(PublishedManager,self).get_queryset().filter(status='published')
class Post(models.Model): 
  # 重新创建初始模型管理器，确保初始模型管理器可⽤
  objects = models.Manager()
  # 使⽤⾃定义模型管理器PublishedManager类创建⼀个管理器实例
  published = PublishedManager()
  # 增加tags字段
  tags = TaggableManager()
  STATUS_CHOICES = ( 
    ('draft', 'Draft'),
    ('published', 'Published') 
  ) 
  title = models.CharField(max_length=250) 
  body = models.TextField() 
  slug = models.SlugField(max_length=250, unique_for_date='publish') 
  author = models.ForeignKey(User, on_delete=models.CASCADE, 
related_name='author') 
  publish = models.DateTimeField(default=timezone.now) 
  created = models.DateTimeField(auto_now_add=True) 
  updated = models.DateTimeField(auto_now=True) 
  status = models.CharField(max_length=10, choices=STATUS_CHOICES, 
default='draft')
  def get_absolute_url(self): 
     return reverse('blog:post_detail', args=[self.publish.year, 
                  self.publish.month, self.publish.day, self.slug])
  
class Meta: 
      ordering = ('-publish',) 
def __str__(self): 
     return self.title
def get_absolute_url(self):
   return reverse('blog:post_detail', args=[self.publish.year,
                  self.publish.month, self.publish.day, self.slug])
#添加评论数据模型类
class Comment(models.Model): 
    post = models.ForeignKey(Post, 
                            on_delete=models.CASCADE, 
                            related_name='comments') 
    name = models.CharField(max_length=80) 
    email = models.EmailField() 
    body = models.TextField() 
    created = models.DateTimeField(auto_now_add=True) 
    updated = models.DateTimeField(auto_now=True) 
    active = models.BooleanField(default=True) 
    class Meta: 
      ordering = ('created',) 
    def __str__(self): 
      return f'Comment by {self.name} on {self.post}'